import tensorflow as tf
import keras as keras


class LinearModel(keras.Model):
    def __init__(self):
        super().__init__()
        self.con = keras.layers.Conv2D(64,kernel_size=(3,3),activation=tf.nn.relu)
        self.maxPool = keras.layers.MaxPool2D(pool_size=(2,2))
        self.dense = keras.layers.Dense(units=1)

    def call(self, inputs):
        x = self.con(inputs)
        x = self.maxPool(x)
        x = self.dense(x)
        return x
    
